import cv2
import numpy as np


# 加载图像
img1 = cv2.imread('cat.jpg')
img2 = cv2.imread('opencv-logo.png')
# I want to put logo on top-left corner, So I create a ROI
rows,cols,channels = img2.shape
roi = img1[0:rows, 0:cols ]
# Now create a mask of logo and create its inverse mask also
img2gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
ret, mask = cv2.threshold(img2gray, 175, 255, cv2.THRESH_BINARY)
# 位取反
mask_inv = cv2.bitwise_not(mask)
# 位与
img1_bg = cv2.bitwise_and(roi,roi,mask = mask)
img2_fg = cv2.bitwise_and(img2,img2,mask = mask_inv)
dst = cv2.add(img1_bg,img2_fg)
# img1[0:rows, 0:cols ] = dst

# print(img1.shape, img2.shape)
# print(img1_bg.shape, img2_fg.shape)

cv2.imshow("img1", img1)
cv2.imshow('img2',img2)
# cv2.imshow('img2gray',img2gray)
# cv2.imshow('mask',mask)
# cv2.imshow('mask_inv',mask_inv)
cv2.imshow('img1_bg',img1_bg)
cv2.imshow('img2_fg',img2_fg)
# cv2.imshow('dst',dst)
# cv2.imshow('img1',img1)
# cv2.imwrite("output.jpg", img2_fg)

cv2.waitKey(0)
cv2.destroyAllWindows()
